Image Statistics Modules¶
from PIL import ImageFile
fp = open("lena.pgm", "rb")
p = ImageFile.Parser()
while 1:
s = fp.read(1024)
if not s:
break
p.feed(s)
im = p.close()
im.save("copy.jpg")
official modes¶
script
"1": ("L", "L", ("1",)),
"L": ("L", "L", ("L",)),
"I": ("L", "I", ("I",)),
"F": ("L", "F", ("F",)),
"P": ("P", "L", ("P",)),
"RGB": ("RGB", "L", ("R", "G", "B")),
"RGBX": ("RGB", "L", ("R", "G", "B", "X")),
"RGBA": ("RGB", "L", ("R", "G", "B", "A")),
"CMYK": ("RGB", "L", ("C", "M", "Y", "K")),
"YCbCr": ("RGB", "L", ("Y", "Cb", "Cr")),
"LAB": ("RGB", "L", ("L", "A", "B")),
"HSV": ("RGB", "L", ("H", "S", "V")),In [1]:
import os
from PIL import Image
from PIL import ImageFile
from PIL.ImageStat import Stat
from PIL.ImageOps import grayscale
im_file_dir = '/Users/mojo/git_clone/dlanier/La_Machina_Fractal_Arien/images'
if os.path.isdir(im_file_dir) == False:
print(im_file_dir, '\nDirectory not found')
In [2]:
os.listdir(im_file_dir)
Out[2]:
In [3]:
FILE_EXT = ['.png']
im_dir = im_file_dir # os.path.abspath('../images')
file_list = os.listdir(im_dir)
png_files = []
for maybe_file in file_list:
_, file_ext = os.path.splitext(maybe_file)
if file_ext in FILE_EXT:
print(maybe_file)
full_file_name = os.path.join(im_dir, maybe_file)
print('\t%s'%full_file_name)
p = ImageFile.Parser()
with open(full_file_name, 'rb') as fp:
while 1:
s = fp.read(1024)
if not s:
break
p.feed(s)
im = p.close()
print('type(im)',type(im))
im_stat = Stat(im)
stddev = im_stat.stddev
print(type(stddev), len(stddev), 'stddev')
for std in stddev:
print(std)
print('\n')
In [4]:
import os
from collections import OrderedDict
from PIL import ImageFile
from PIL.ImageStat import Stat
import pandas as pd
def get_image_file_stats(imfilename, imfiledirectory=None):
""" stats_dict, image_input = get_image_file_stats(imfilename, imfiledirectory) """
stats_dict = {}
full_file_name = os.path.join(imfiledirectory, imfilename)
if os.path.isfile(full_file_name):
p = ImageFile.Parser()
with open(full_file_name, 'rb') as fp:
while 1:
s = fp.read(1024)
if not s:
break
p.feed(s)
image_input = p.close()
stats_dict = get_image_stats(image_input)
stats_dict['image_file_name'] = imfilename
else:
print('Failed to open file:', full_file_name)
return stats_dict, image_input
def get_image_stats(image_input):
stats_dict = {}
im_stat = Stat(im)
stats_dict['extrema'] = im_stat.extrema
stats_dict['count'] = im_stat.count
stats_dict['sum'] = im_stat.sum
stats_dict['sum2'] = im_stat.sum2
stats_dict['mean'] = im_stat.mean
stats_dict['median'] = im_stat.median
stats_dict['rms'] = im_stat.rms
stats_dict['var'] = im_stat.var
stats_dict['stddev'] = im_stat.stddev
return stats_dict
def show_image_stats(imfilename, imfiledirectory):
if imfiledirectory is None:
imfiledirectory = os.getcwd()
stats_dict, image_input = get_image_file_stats(imfilename, imfiledirectory)
for k, v in stats_dict.items():
print('%20s: '%(k), v)
display(image_input)
def get_images_stats_dataframe(images_dir_list, file_extensions_list):
"""images_stats_dataframe = get_images_stats_dataframe(images_dir_list, file_extensions_list) """
STATS_KEYS = ["count","extrema","image_file_name","mean","median","rms","stddev","sum","sum2","var"]
images_stats_dataframe = None
stats_dict_list = []
for maybe_image_dir in images_dir_list:
if os.path.isdir(maybe_image_dir):
files_list = os.listdir(maybe_image_dir)
for maybe_file in files_list:
_, file_ext = os.path.splitext(maybe_file)
if file_ext in file_extensions_list:
stats_dict, _ = get_image_file_stats(maybe_file, maybe_image_dir)
stats_dict_list.append(stats_dict)
images_stats_dataframe = pd.DataFrame(stats_dict_list)
return images_stats_dataframe
In [5]:
imfiledir = '/Users/mojo/git_clone/dlanier/La_Machina_Fractal_Arien/images'
imfilename = 'helloKeras.png'
stats_dict, _ = get_image_file_stats(imfilename, imfiledir)
s = '"'
for k in sorted(list(stats_dict.keys())):
s += k + '","'
print('[',s[:-2] ,']')
In [6]:
FILE_EXT = ['.png']
# im_dir = os.path.abspath('../images')
images_stats_dataframe = get_images_stats_dataframe([im_dir], FILE_EXT)
images_stats_dataframe
Out[6]:
In [7]:
FILE_EXT = ['.png']
# im_dir = os.path.abspath('../images')
file_list = os.listdir(im_dir)
png_files = []
for maybe_file in file_list:
_, file_ext = os.path.splitext(maybe_file)
if file_ext in FILE_EXT:
show_image_stats(maybe_file, im_dir)
